//
// Created by jack on 2022/3/9.
//
#include "usermodel.hpp"
#include "db.hpp"

bool UserModel::insert(User &user)
{
    // 1. 组装MySQL语句
    char sql[1024]{0};
    std::sprintf(sql, "insert into user(name,password,state) value('%s','%s','%s')", user.getName().c_str(),
                 user.getPassword().c_str(), user.getState().c_str());
    MySQL mysql;
    if (mysql.connect())
    {
        if (mysql.update(sql))
        {
            // 获取插入成功的用户的主键id
            user.setId(mysql_insert_id(mysql.getConnection()));
            return true;
        }
    }
    return false;
}

User UserModel::query(int id)
{
    // 1. 组装MySQL语句
    char sql[1024]{0};
    std::sprintf(sql, "select * from user where id = %d", id);
    MySQL mysql;
    if (mysql.connect())
    {
        auto *res = mysql.query(sql);
        if (res != nullptr)
        {
            MYSQL_ROW row = mysql_fetch_row(res);
            if (row != nullptr)
            {
                User user;
                user.setId(atoi(row[0]));
                user.setName(row[1]);
                user.setPassword(row[2]);
                user.setState(row[3]);
                mysql_free_result(res);
                return user;
            }
        }
    }
    return User();
}

bool UserModel::updateState(User user)
{
    char sql[1024]{0};
    std::sprintf(sql, "update user set state = '%s' where id = %d", user.getState().c_str(), user.getId());
    MySQL mysql;
    if (mysql.connect())
    {
        if (mysql.update(sql))
        {
            return true;
        }
    }
    return false;
}

void UserModel::resetState()
{
    char sql[1024]{0};
    std::sprintf(sql, "update user set state = 'offline' where state = 'online'");
    MySQL mysql;
    if (mysql.connect())
    {
        mysql.update(sql);

    }

}
